package com.tt.utils;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import javax.websocket.Session;

public class WebSocketUtils {
	
	private static Map<String,Session> map = new ConcurrentHashMap<String,Session>();


    public static void put(String userid,Session session){
        map.put(userid,session);
    }


    public static Session get(String userid){
        return map.get(userid);

    }

    public static List<Session> getOtherSession(String userid){
        List<Session> result = new ArrayList<Session>();
        Set<Map.Entry<String, Session>> set=  map.entrySet();
//        Iterator<Map.Entry<String, Session>> iterator= set.iterator();
        for(Map.Entry<String, Session> s:set){
            if(!s.getKey().equals(userid)){
                result.add(s.getValue());
            }
        }
        return result;
    }

    public static void remove(String userid){
        map.remove(userid);
    }

    public static boolean hasConnection(String userid){
        return map.containsKey(userid);
    }
}
